package com.yc.cloud.openai.entity;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableName;
import com.yc.cloud.common.orm.BaseEntity;

import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;

/**
 * 问答知识库
 * 
 * @author Yancy
 * @since 2025-01-07
 */
@EqualsAndHashCode(callSuper = true)
@Data
@TableName("gpt_dataset")
@ApiModel(value = "GptDataset对象", description = "问答知识库")
public class GptDataset extends BaseEntity {

    private static final long serialVersionUID = 1L;


    @ApiModelProperty("知识库ID")
    @TableField("dataset_id")
    private String datasetId;

    @ApiModelProperty("名称")
    @TableField("name")
    private String name;

    @ApiModelProperty("描述")
    @TableField("description")
    private String description;

    @ApiModelProperty("父ID")
    @TableField("parent_id")
    private Long parentId;

    @ApiModelProperty("封面ID")
    @TableField("cover")
    private String cover;

    @ApiModelProperty("向量模型")
    @TableField("vector_model")
    private String vectorModel;

    @ApiModelProperty("大模型")
    @TableField("agent_model")
    private String agentModel;

    public static final String NAME = "name";

    public static final String DESCRIPTION = "description";

    public static final String PARENT_ID = "parent_id";

    public static final String COVER = "cover";

    public static final String VECTOR_MODEL = "vector_model";

    public static final String AGENT_MODEL = "agent_model";

}
